transformers >= 4.31.0
torch >= 2.0.0
flash-attn >= 2.3.3
datasets >= 2.14.0
nltk >= 3.8.0
sentencepiece >= 0.1.0
git+https://github.com/facebookresearch/llama.git
